package leetCode;

import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;

/**
 有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。
 给你一个坐标 (sr, sc) 表示图像渲染开始的像素值（行 ，列）和一个新的颜色值 newColor，让你重新上色这幅图像。
 为了完成上色工作，从初始坐标开始，记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点，
 接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点，……，
 重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
 最后返回经过上色渲染后的图像.
 输入: image = [[1,1,1],[1,1,0],[1,0,1]]
 sr = 1, sc = 1, newColor = 2
 输出: [[2,2,2],[2,2,0],[2,0,1]]
 1 1 1       2 2 2
 1 1 0 -->   2 2 0
 1 0 1       2 0 1     两临且数字相同的块，变为2，一直持续，直到无法改变
 解析:
 在图像的正中间，(坐标(sr,sc)=(1,1)),
 在路径上所有符合条件的像素点的颜色都被更改成2。
 注意，右下角的像素没有更改为2，
 因为它不是在上下左右四个方向上与初始点相连的像素点。
 * @Author: chen jinxu
 * @Date: 2021/11/21 17:59
 * @Description: //TODO
 */
public class LeetCode733 {
    public static void main(String[] args) {
        int[][] image = {{1,1,1},{1,1,0},{1,0,1}};
        int sr = 1;
        int sc = 1;
        int newColor = 2;
        int[][] result = floodFill(image, sr, sc, newColor);
        int length = result.length;
        for (int i = 0; i < length; i++) {
            System.out.println(Arrays.toString(result[i]));
        }
    }
    // 定义上下左右的坐标变化
    public  static int[] dx = {1, 0, 0, -1};
    public static int[] dy = {0, 1, -1, 0};

    public static int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        int currColor = image[sr][sc];
        if (currColor == newColor) {
            return image;
        }
        // 获取行的数量
        int n = image.length;
        //获取列的数量，既每行多少个元素
        int m = image[0].length;
        Queue<int[]> queue = new LinkedList<>();
        queue.offer(new int[]{sr, sc});
        image[sr][sc] = newColor;
        while (!queue.isEmpty()) {
            int[] cell = queue.poll();
            int x = cell[0], y = cell[1];
            for (int i = 0; i < 4; i++) {
                int mx = x + dx[i], my = y + dy[i];
                if (mx > 0 && mx < n && my > 0 && my < m && image[mx][my] == currColor) {
                    queue.offer(new int[]{mx, my});
                    image[mx][my] = newColor;
                }
            }
        }
        return image;
    }


}
